package com.intretech.umsin.system.service.impl;

import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.intretech.umsin.datasource.dao.KernDAO;
import com.intretech.umsin.security.satoken.utils.LoginHelper;
import com.intretech.umsin.system.converter.SysRoleVoEntityConverter;
import com.intretech.umsin.system.entity.SysRoleEntity;
import com.intretech.umsin.system.mapper.SysRoleMapper;
import com.intretech.umsin.system.param.SysRoleParam;
import com.intretech.umsin.system.service.SysRoleService;
import com.intretech.umsin.system.vo.SysRoleVo;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.io.Serializable;
import java.util.*;


/**
* @version V1.0
* 角色信息:sys_role 对应商业服务实现.
* @ClassName: SysRoleServiceImpl
* @Description: 角色信息:sys_role 对应商业服务实现.
* @author: 李启联
* @date:   2022-8-6 14:37:22
* @Copyright: Copyright (c) 2022
*/
@Service
public class SysRoleServiceImpl extends ServiceImpl<SysRoleMapper, SysRoleEntity> implements SysRoleService {

    @Autowired
    private SysRoleVoEntityConverter sysRoleVoEntityConverter;

    @Autowired
    KernDAO kernDAO;


    /**
    * 根据ID查询单个角色信息对象
    * @param id 组键ID
    * @return 单个角色信息VO对象
    */
    public SysRoleVo queryById(Serializable id){
        if (id == null) {return null;}
        return sysRoleVoEntityConverter.reverseFromTarget(this.getById(id));
    }


    /**
    * 根据ID集合查询返回角色信息列表
    * @param idList 组键ID集合
    * @return 角色信息列表
    */
    public List<SysRoleVo> queryByIds(Collection<? extends Serializable> idList){
        if(CollectionUtils.isEmpty(idList)){return null;}
        return sysRoleVoEntityConverter.reverseFromTarget(this.listByIds(idList));
    }


    /**
    * 分页查询 角色信息 信息
    * @param page 分页对象
    * @param param 查询参数
    * @return IPage<SysRoleVo> 分页对象
    */
    @Override
    public IPage<SysRoleVo> queryForPage(IPage page,SysRoleParam param){
      IPage<SysRoleEntity> page0 = this.page(page,createQueryWrapper(param));
      IPage<SysRoleVo> pageResult = page0.convert(e -> sysRoleVoEntityConverter.reverseFromTarget(e));
      return pageResult;
    }


    /**
    * 根据条件查询返回 角色信息 对象列表
    * @param  param 查询参数
    * @return 角色信息 对象列表
    */
    @Override
    public List<SysRoleVo> queryForList(SysRoleParam param) {
        List<SysRoleEntity> list = this.list(createQueryWrapper(param));
        return sysRoleVoEntityConverter.reverseFromTarget(list);
    }

    /**
    * 构建查询条件
    * @param param 查询参数实体对象
    * @return 构建后的QueryWrapper查询条件
    */
    private QueryWrapper<SysRoleEntity> createQueryWrapper(SysRoleParam param){
        QueryWrapper<SysRoleEntity> queryWrapper = new QueryWrapper<>();
        if(ObjectUtils.isEmpty(param)) { return queryWrapper; }
        Optional<SysRoleParam> op = Optional.of(param);
        queryWrapper.lambda()
                .eq(op.map(SysRoleParam::getId).isPresent(),SysRoleEntity::getId,param.getId())
                .eq(op.map(SysRoleParam::getRoleName).filter(StringUtils::isNotBlank).isPresent(),SysRoleEntity::getRoleName,param.getRoleName())
                .eq(op.map(SysRoleParam::getRoleKey).filter(StringUtils::isNotBlank).isPresent(),SysRoleEntity::getRoleKey,param.getRoleKey())
                .eq(op.map(SysRoleParam::getRoleSort).isPresent(),SysRoleEntity::getRoleSort,param.getRoleSort())
                .eq(op.map(SysRoleParam::getDataScope).filter(StringUtils::isNotBlank).isPresent(),SysRoleEntity::getDataScope,param.getDataScope())
                .eq(op.map(SysRoleParam::getStatus).filter(StringUtils::isNotBlank).isPresent(),SysRoleEntity::getStatus,param.getStatus())
                .eq(op.map(SysRoleParam::getRemark).filter(StringUtils::isNotBlank).isPresent(),SysRoleEntity::getRemark,param.getRemark())
        ;
        return queryWrapper;
    }

    /**
    * 添加角色信息到数据库
    * @param vo  角色信息 VO对象
    * @return 添加后的角色信息 VO对象
    */
    @Override
    public SysRoleVo add(SysRoleVo vo) {
        if (ObjectUtils.isEmpty(vo)) {return null;}
        SysRoleEntity entity = sysRoleVoEntityConverter.convertToTarget(vo);
        if(this.save(entity)){
            return sysRoleVoEntityConverter.reverseFromTarget(this.getById(entity.getId()));
        }else {
            throw new RuntimeException("添加 角色信息 单个对象失败!");
            //return null;
        }
    }

    /**
    * 添加角色信息集合列表到数据库
    * @param vos  角色信息 VO对象列表
    * @return 添加后的角色信息 VO对象集合列表
    */
    @Override
    public List<SysRoleVo> adds(List<SysRoleVo> vos) {
        if (CollectionUtils.isEmpty(vos)) { return null; }
        List<SysRoleEntity> entities = sysRoleVoEntityConverter.convertToTarget(vos);
        if(this.saveBatch(entities)){
            List<Serializable> ids = new ArrayList<>();
            entities.forEach(e->{ids.add(e.getId());});
            return sysRoleVoEntityConverter.reverseFromTarget(this.listByIds(ids));
        }else {
           throw new RuntimeException("添加 角色信息 集合列表失败!");
           //return null;
        }
    }

    /**
    * 更新角色信息到数据库
    * @param vo 要更新的角色信息 VO对象
    * @return 更新后的角色信息 VO对象
    */
    @Override
    public SysRoleVo modify(SysRoleVo vo) {
        if (ObjectUtils.isEmpty(vo)) {return null;}
        SysRoleEntity entity = sysRoleVoEntityConverter.convertToTarget(vo);
        if(this.updateById(entity)){
            return sysRoleVoEntityConverter.reverseFromTarget(this.getById(entity.getId()));
        }else {
            throw new RuntimeException("更新角色信息单个对象失败!");
            //return null;
        }
    }

    /**
    * 更新角色信息集合列表到数据库
    * @param vos 要更新的角色信息 VO对象集合
    * @return 更新后的角色信息 VO对象集合
    */
    @Override
    public List<SysRoleVo> modifys(List<SysRoleVo> vos) {
        if (CollectionUtils.isEmpty(vos)) { return null; }
        List<SysRoleEntity> entities = sysRoleVoEntityConverter.convertToTarget(vos);
        if(this.updateBatchById(entities)){
            List<Serializable> ids = new ArrayList<>();
            entities.forEach(e->{ids.add(e.getId());});
            return sysRoleVoEntityConverter.reverseFromTarget(this.listByIds(ids));
        }else {
            throw new RuntimeException("更新角色信息集合列表失败!");
            //return null;
        }
    }

    /**
     * 根据用户ID查询角色列表
     *
     * @param userId 用户ID
     * @return 角色列表
     */
    @Override
    public List<SysRoleVo> selectRolesByUserId(Long userId)  {
        List<SysRoleVo> sysRoles = null;
        try {
            sysRoles = kernDAO.queryForList("select * from sys_role where id in (select role_id from sys_user_role where user_id = ?)", SysRoleVo.class, userId);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return sysRoles;
    }

    /**
     * 根据用户ID查询角色权限 (获取角色数据权限)
     *
     * @param userId 用户ID
     * @return 权限列表
     */
    @Override
    public Set<String> selectRolePermissionByUserId(Long userId) {
        Set<String> permsSet = new HashSet<String>();
        // 管理员拥有所有权限
        if (LoginHelper.isAdmin(userId)) {
            permsSet.add("admin");
        } else {
            List<SysRoleVo> perms = selectRolesByUserId(userId);
            for (SysRoleVo perm : perms) {
                if (ObjectUtil.isNotNull(perm)) {
                    permsSet.addAll(Arrays.asList(perm.getRoleKey().trim().split(",")));
                }
            }
        }
        return permsSet;
    }

}

